class VelocityModel:
    def __init__(self):
        pass

    def get_position(self, ego_vehicle):
        pos = ego_vehicle.pos
        vel = ego_vehicle.vel
        new_pos = [pos[0] + vel[0] * 0.1, pos[1] + vel[1] * 0.1]
        return new_pos

    def get_position_by_state(self, self_state):
        pos = [self_state[3], self_state[4]]
        vel = [self_state[10], self_state[11], self_state[12]]
        new_pos = [pos[0] + vel[0] * 0.1, pos[1] + vel[1] * 0.1]
        return new_pos

    def get_position_distribution(self, ego_vehicle):
        pass
